// LeetCode 主站 Problem Nr. 103: 二叉树的锯齿形层序遍历

/*
给你二叉树的根节点 root ，返回其节点值的 锯齿形层序遍历 。
（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

示例 1：
	输入：root = [3,9,20,null,null,15,7]
	输出：[[3],[20,9],[15,7]]
示例 2：
	输入：root = [1]
	输出：[[1]]
示例 3：
	输入：root = []
	输出：[]

提示：
    树中节点数目在范围 [0, 2000] 内
    -100 <= Node.val <= 100

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

package main

import "github.com/saint-yellow/think-leetcode/ds"

type TreeNode = ds.BinaryNode[int]

func zigzagLevelOrder(root *TreeNode) [][]int {
	return method1(root)
}

func method1(root *TreeNode) [][]int {
	result := make([][]int, 0)
	if root == nil {
		return result
	}

	queue := []*TreeNode{root}
	level := 0
	for len(queue) > 0 {
		size := len(queue)
		nodes := make([]int, size)
		for i := 0; i < size; i++ {
			node := queue[0]
			queue = queue[1:]
			if node.Left != nil {
				queue = append(queue, node.Left)
			}
			if node.Right != nil {
				queue = append(queue, node.Right)
			}
		}

		// 将来自奇数层的数组的元素反转
		if level % 2 != 0 {
			for i, j := 0, len(nodes)-1; i < j; i, j = i+1, j-1 {
				nodes[i], nodes[j] = nodes[j], nodes[i]
			}
		}

		// 本层处理完成，将本层结果加入到总结果
		level++
		result = append(result, nodes)
	}
	return result
}
